from base import base
class Shift(base):
    def __init__(self, cfg) -> None:
        super(Shift, self).__init__(cfg)
        self.columns = ["uid", "user_city", "item_id", "author_id", "item_city", "channel", "finish", "like", "music_id", "did", "create_time", "video_duration"]
    def shfit_min_0(self, line):
        new_line = []
        for v, name in zip(line, self.columns):
            if v == -1:
                v = self.dataset_cfg.max_val(name)-self.dataset_cfg.max_val(name) # 数组最大索引
            new_line.append(v)
        return new_line

    
        